Lambda functions in python using VS code